multiclass classification using tensorflow js